package com.aerors.Util;

import java.io.BufferedReader;
import java.io.DataInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.HttpURLConnection;
import java.net.URL;
import java.net.URLConnection;
import java.nio.charset.Charset;
import java.util.List;
import java.util.Map;

import org.apache.commons.lang3.StringUtils;

public class ElasticSearchTest {

	public static void main(String[] args) {
//		String url = "http://localhost:9200/_count";
//		String param = "pretty";
//
//		String result = doGet(url, param);
//
//		System.out.println(result);
		
		String url = "http://localhost:9200/accounts";
		String data = "{\r\n" + 
				"  \"mappings\": {\r\n" + 
				"    \"person\": {\r\n" + 
				"      \"properties\": {\r\n" + 
				"        \"user\": {\r\n" + 
				"          \"type\": \"text\",\r\n" + 
				"          \"analyzer\": \"ik_max_word\",\r\n" + 
				"          \"search_analyzer\": \"ik_max_word\"\r\n" + 
				"        },\r\n" + 
				"        \"title\": {\r\n" + 
				"          \"type\": \"text\",\r\n" + 
				"          \"analyzer\": \"ik_max_word\",\r\n" + 
				"          \"search_analyzer\": \"ik_max_word\"\r\n" + 
				"        },\r\n" + 
				"        \"desc\": {\r\n" + 
				"          \"type\": \"text\",\r\n" + 
				"          \"analyzer\": \"ik_max_word\",\r\n" + 
				"          \"search_analyzer\": \"ik_max_word\"\r\n" + 
				"        }\r\n" + 
				"      }\r\n" + 
				"    }\r\n" + 
				"  }\r\n" + 
				"}";
		String result = doPut(url, data, "", null);
		System.out.println(result);
	}

	public static String doGet(String url, String param) {
		String result = "";
		BufferedReader in = null;
		try {
			String urlNameString = url + "?" + param;
			URL realUrl = new URL(urlNameString);
			// 打开和URL之间的连接
			URLConnection connection = realUrl.openConnection();
			// 设置通用的请求属性
			connection.setRequestProperty("accept", "*/*");
			connection.setRequestProperty("connection", "Keep-Alive");
			connection.setRequestProperty("user-agent", "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1;SV1)");
			// 建立实际的连接
			connection.connect();
			// 获取所有响应头字段
			Map<String, List<String>> map = connection.getHeaderFields();
			// 遍历所有的响应头字段
			for (String key : map.keySet()) {
				System.out.println(key + "--->" + map.get(key));
			}
			// 定义 BufferedReader输入流来读取URL的响应
			in = new BufferedReader(new InputStreamReader(connection.getInputStream()));
			String line;
			while ((line = in.readLine()) != null) {
				result += line;
			}
		} catch (Exception e) {
			System.out.println("发送GET请求出现异常！" + e);
			e.printStackTrace();
		}
		// 使用finally块来关闭输入流
		finally {
			try {
				if (in != null) {
					in.close();
				}
			} catch (Exception e2) {
				e2.printStackTrace();
			}
		}
		return result;
	}

	public static String doPost(String url, String param) {
		PrintWriter out = null;
		BufferedReader in = null;
		String result = "";
		try {
			URL realUrl = new URL(url);
			// 打开和URL之间的连接
			URLConnection conn = realUrl.openConnection();
			// 设置通用的请求属性
			conn.setRequestProperty("accept", "*/*");
			conn.setRequestProperty("connection", "Keep-Alive");
			conn.setRequestProperty("user-agent", "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1;SV1)");
			// 发送POST请求必须设置如下两行
			conn.setDoOutput(true);
			conn.setDoInput(true);
			// 获取URLConnection对象对应的输出流
			out = new PrintWriter(conn.getOutputStream());
			// 发送请求参数
			out.print(param);
			// flush输出流的缓冲
			out.flush();
			// 定义BufferedReader输入流来读取URL的响应
			in = new BufferedReader(new InputStreamReader(conn.getInputStream()));
			String line;
			while ((line = in.readLine()) != null) {
				result += line;
			}
		} catch (Exception e) {
			System.out.println("发送 POST 请求出现异常！" + e);
			e.printStackTrace();
		}
		// 使用finally块来关闭输出流、输入流
		finally {
			try {
				if (out != null) {
					out.close();
				}
				if (in != null) {
					in.close();
				}
			} catch (IOException ex) {
				ex.printStackTrace();
			}
		}
		return result;
	}

	public static String doPut(String urlPath, String data, String charSet, String[] header) {
		String result = null;
		URL url = null;
		HttpURLConnection httpurlconnection = null;
		try {
			url = new URL(urlPath);
			httpurlconnection = (HttpURLConnection) url.openConnection();
			httpurlconnection.setDoInput(true);
			httpurlconnection.setDoOutput(true);
			httpurlconnection.setConnectTimeout(2000000);// 设置连接主机超时（单位：毫秒）
			httpurlconnection.setReadTimeout(2000000);// 设置从主机读取数据超时（单位：毫秒）

			if (header != null) {
				for (int i = 0; i < header.length; i++) {
					String[] content = header[i].split(":");
					httpurlconnection.setRequestProperty(content[0], content[1]);
				}
			}

			httpurlconnection.setRequestMethod("PUT");
			httpurlconnection.setRequestProperty("Content-Type", "application/x-www-form-urlencoded");

			if (StringUtils.isNotBlank(data)) {
				httpurlconnection.getOutputStream().write(data.getBytes("UTF-8"));
			}
			httpurlconnection.getOutputStream().flush();
			httpurlconnection.getOutputStream().close();
			int code = httpurlconnection.getResponseCode();

			System.out.println("response code :" + code);
			DataInputStream in = new DataInputStream(httpurlconnection.getInputStream());
			int len = in.available();
			byte[] by = new byte[len];
			in.readFully(by);
			if (StringUtils.isNotBlank(charSet)) {
				result = new String(by, Charset.forName(charSet));
			} else {
				result = new String(by);
			}
			in.close();
//			if (code == 200) {
//				
//			} else {
//				System.out.println("请求地址：" + urlPath + "返回状态异常，异常号为：" + code);
//			}
		} catch (Exception e) {
			System.out.println("访问url地址：" + urlPath + "发生异常" + e.toString());
		} finally {
			url = null;
			if (httpurlconnection != null) {
				httpurlconnection.disconnect();
			}
		}
		return result;
	}

	public static void doDelete() {

	}
}
